import tensorflow as tf
from keras.layers import Dense
import os

if __name__ == '__main__':
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
    layer = Dense(units=2, activation='relu')

    X = tf.random.normal([1, 3])
    print(X)
    output = layer(X)
    print(output)
    # 查看当前的权重
    print(layer.kernel)
    print(layer.bias)
